package com.zhang.common.core.utils.reflect;

import cn.hutool.core.convert.Convert;
import com.zhang.common.core.utils.DateUtils;
import org.apache.commons.lang3.StringUtils;
import org.apache.commons.lang3.Validate;
import org.apache.poi.ss.usermodel.DateUtil;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.lang.reflect.*;
import java.util.Arrays;
import java.util.Date;

/**
 * @Class com.zhang.common.core.utils.reflect
 * @Author zhang
 * @Date 2025/9/11 14:14
 * @Description 反射工具类.
 * 提供调用getter/setter方法, 访问私有变量, 调用私有方法
 * 获取泛型类型Class, 被AOP过的真实类等工具函数.
 **/
public class ReflectUtils {

    /**
     * set 属性前缀
     */
    private static final String SETTER_PREFIX = "set";

    /**
     * get 属性前缀
     */
    private static final String GETTER_PREFIX = "get";

    /**
     * 父类对象包含指定的字符
     */
    private static final String CGLIB_CLASS_SEPARATOR = "$$";

    /**
     * 日志打印对象
     */
    private static final Logger logger = LoggerFactory.getLogger(ReflectUtils.class);

    /**
     * 调用对象的 Getter方法
     * 支持多级，如：对象名.对象名.方法
     *
     * @param obj 实际对象
     * @param propertyName 对应的属性
     * @param <E> 泛型
     * @return 返回对应属性 getter 方法的返回值
     */
    @SuppressWarnings("unchecked")
    public static <E> E invokeGetter(Object obj, String propertyName) {
        Object object = obj;
        // 对应的属性名进行指定分隔符分割获取对应的数据集
        for (String name : StringUtils.split(propertyName, ".")) {
            // 获取对应的指定方法名
            String getterMethodName = GETTER_PREFIX + StringUtils.capitalize(name);
            // 调用指定方法获取对应的返回值
            object = invokeMethod(object, getterMethodName, new Class[]{}, new Object[]{});
        }
        return (E) object;
    }

    /**
     * 调用 Setter 方法, 仅匹配方法名
     * 支持多级，如：对象名.对象名.方法
     *
     * @param obj 实际对象
     * @param propertyName 对应的属性
     * @param value 设置的属性值
     * @param <E> 泛型
     */
    public static <E> void invokeSetter(Object obj, String propertyName, E value) {
        Object object = obj;
        // 对应的属性名进行指定分隔符分割获取对应的数据集
        String[] names = StringUtils.split(propertyName, ".");
        // 遍历对应的数据集
        for (int i = 0; i < names.length; i++) {
            // 不为最后一个数据
            if (i < names.length - 1) {
                // 获取对应的 getter 方法
                String getterMethodName = GETTER_PREFIX + StringUtils.capitalize(names[i]);
                // 根据 getter 方法获取对应的对象类
                object = invokeMethod(object, getterMethodName, new Class[]{}, new Object[]{});
            } else {
                // 获取对应的 setter 方法
                String setterMethodName = SETTER_PREFIX + StringUtils.capitalize(names[i]);
                // 调用对应属性的 setter 方法设置对应值
                invokeMethodByName(object, setterMethodName, new Object[]{value});
            }
        }
    }

    /**
     * 直接读取对象属性值, 无视 private/protected 修饰符, 不经过 getter 函数.
     *
     * @param obj 实际对象
     * @param fieldName 属性名称
     * @param <E> 泛型
     * @return 返回对应的属性方法返回值
     */
    @SuppressWarnings("unchecked")
    public static <E> E getFieldValue(final Object obj, final String fieldName) {
        // 获取对应的属性
        Field field = getAccessibleField(obj, fieldName);
        if (field == null) {
            logger.info("在 {} 中，没有找到 {} 字段 ", obj.getClass(), fieldName);
            return null;
        }
        E result = null;
        try {
            // 获取对应属性对象的属性值
            result = (E) field.get(obj);
        } catch (IllegalAccessException e) {
            logger.error("不可能抛出的异常{}", e.getMessage(), e);
        }
        return result;
    }

    /**
     * 直接设置对象属性值, 无视 private/protected 修饰符, 不经过setter函数
     *
     * @param obj 实际对象
     * @param fieldName 属性名称
     * @param value 属性值
     * @param <E> 泛型
     */
    public static <E> void setFieldValue(final Object obj, final String fieldName, final E value) {
        // 获取对应的属性
        Field field = getAccessibleField(obj, fieldName);
        if (field == null) {
            logger.info("在 {} 中，没有找到 {} 字段 ", obj.getClass(), fieldName);
            return;
        }
        try {
            // 设置对应的属性值
            field.set(obj, value);
        } catch (IllegalAccessException e) {
            logger.error("不可能抛出的异常: {}", e.getMessage(), e);
        }
    }

    /**
     * 直接调用对象方法, 无视 private/protected 修饰符
     * 用于一次性调用的情况, 否则应使用 getAccessibleMethod() 函数获得 Method 后反复调用
     * 同时匹配方法名 + 参数类型
     *
     * @param obj 实际对象
     * @param methodName 调用的方法名称
     * @param parameterTypes 实际参数类型
     * @param args 参数数据集
     * @param <E> 泛型
     * @return  返回执行方法后的返回值
     */
    @SuppressWarnings("unchecked")
    public static <E> E invokeMethod(final Object obj, final String methodName,
                                     final Class<?>[] parameterTypes, final Object[] args) {
        // 实际对象|调用方法判空
        if (obj == null || methodName == null) {
            return null;
        }
        // 获取对应的实际对象的方法
        Method method = getAccessibleMethod(obj, methodName, parameterTypes);
        // 方法对象判空
        if (method == null) {
            logger.info("在 {} 中，没有找到 {} 方法 ", obj.getClass(), methodName);
            return null;
        }
        try {
            return (E) method.invoke(obj, args);
        } catch (Exception e) {
            logger.error("对象 {} 调用对应的属性方法 {} 返回值异常: {}", obj.getClass(), methodName, e.getMessage(), e);
            String msg = "method: " + method + ", obj: " + obj + ", args: " + Arrays.toString(args);
            throw convertReflectionExceptionToUnchecked(msg, e);
        }
    }

    /**
     * 直接调用对象方法, 无视private/protected修饰符，
     * 用于一次性调用的情况，否则应使用getAccessibleMethodByName()函数获得Method后反复调用.
     * 只匹配函数名，如果有多个同名函数调用第一个。
     *
     * @param obj 实际对象
     * @param methodName 属性方法名称
     * @param args 参数数据集
     */
    public static void invokeMethodByName(final Object obj, final String methodName, final Object[] args) {
        // 获取对应的属性方法对象
        Method method = getAccessibleMethodByName(obj, methodName, args.length);
        // 属性方法对象判空
        if (method == null) {
            // 如果为空不报错, 直接返回
            logger.info("在 {} 中，没有找到 {} 方法 ", obj.getClass(), methodName);
            return;
        }
        try {
            // 类型转换 - 将参数数据类型转换为目标方法参数类型
            Class<?>[] cs = method.getParameterTypes();
            // 遍历参数类型对象数据集
            for (int i = 0; i < cs.length; i++) {
                // 参数不为空并且参数类型对象与当前的目标方法参数类型对象不一致时
                if (args[i] != null && !args[i].getClass().equals(cs[i])) {
                    // 目标方法参数类型对象为字符串
                    if (cs[i] == String.class) {
                        // 将对应的参数转为字符串数据
                        args[i] = Convert.toStr(args[i]);
                        // 判断是否以 .0 结尾
                        if (StringUtils.endsWith((String) args[i], ".0")) {
                            // 截取 .0 之前数据
                            args[i] = StringUtils.substringBefore((String) args[i], ".0");
                        }
                    } else if (cs[i] == Integer.class) {
                        // 整形数据
                        args[i] = Convert.toInt(args[i]);
                    } else if (cs[i] == Long.class) {
                        // 长整型数据
                        args[i] = Convert.toLong(args[i]);
                    } else if (cs[i] == Double.class) {
                        // 双精度浮点数据
                        args[i] = Convert.toDouble(args[i]);
                    } else if (cs[i] == Float.class) {
                        // 单精度浮点数据
                        args[i] = Convert.toFloat(args[i]);
                    } else if (cs[i] == Date.class) {
                        // 日期数据
                        if (args[i] instanceof String) {
                            args[i] = DateUtils.parseDate(args[i]);
                        } else {
                            args[i] = DateUtil.getJavaDate((Double) args[i]);
                        }
                    } else if (cs[i] == boolean.class || cs[i] == Boolean.class) {
                        // 布尔数据
                        args[i] = Convert.toBool(args[i]);
                    }
                }
            }
            // 执行对应的属性方法
            method.invoke(obj, args);
        } catch (Exception e) {
            String msg = "method: " + method + ", obj: " + obj + ", args: " + Arrays.toString(args);
            logger.error("对象 {} 对应的属性方法 {} 设置对应的参数 {} 异常: {}", obj.getClass(), methodName, args, e.getMessage(), e);
            throw convertReflectionExceptionToUnchecked(msg, e);
        }
    }

    /**
     * 循环向上转型, 获取对象的DeclaredField, 并强制设置为可访问.
     * 如向上转型到Object仍无法找到, 返回null.
     *
     * @param obj 实际对象
     * @param fieldName 属性名称
     * @return 返回指定的属性名称对应的属性对象
     */
    public static Field getAccessibleField(final Object obj, final String fieldName) {
        // 为空时直接返回 null
        if (obj == null) {
            return null;
        }
        // 校验属性名称
        Validate.notBlank(fieldName, "属性名称不能为空! (FieldName can't be blank!)");
        // 遍历对象的对象类 - 获取对应的父类对象类
        for (Class<?> superClass = obj.getClass(); superClass != Object.class; superClass = superClass.getSuperclass()) {
            try {
                // 获取对应对象声明的对象属性对象
                Field field = superClass.getDeclaredField(fieldName);
                // 强制设置为可访问
                makeAccessible(field);
                return field;
            } catch (NoSuchFieldException e) {
                logger.error("没有对应的属性名称对象！{}", e.getMessage(), e);
            }
        }
        return null;
    }

    /**
     * 循环向上转型, 获取对象的 DeclaredMethod, 并强制设置为可访问
     * 如向上转型到 Object 仍无法找到, 返回null.
     * 匹配函数名 + 参数类型
     * 用于方法需要被多次调用的情况.
     * 先使用本函数先取得 Method, 然后调用 Method.invoke(Object obj, Object... args)
     *
     * @param obj 实际对象
     * @param methodName 调用的属性方法名称
     * @param parameterTypes 参数类型
     * @return 返回对应对象的方法
     */
    public static Method getAccessibleMethod(final Object obj, final String methodName, final Class<?>... parameterTypes) {
        // 为空时- 直接返回 null
        if (obj == null) {
            return null;
        }
        // 校验方法名称
        Validate.notBlank(methodName, "方法名称不能为空！(MethodName can't be blank!)");
        // 遍历对象的对象类 - 获取对应的父类对象类
        for (Class<?> searchType = obj.getClass(); searchType != Object.class; searchType = searchType.getSuperclass()) {
            try {
                // 获取对应的声明方法
                Method method = searchType.getDeclaredMethod(methodName, parameterTypes);
                // 强制设置为可访问
                makeAccessible(method);
                return method;
            } catch (NoSuchMethodException e) {
                logger.error("没有对应的属性方法！{}", e.getMessage(), e);
            }
        }
        return null;
    }

    /**
     * 循环向上转型, 获取对象的DeclaredMethod,并强制设置为可访问.
     * 如向上转型到Object仍无法找到, 返回null.
     * 只匹配函数名。
     * 用于方法需要被多次调用的情况. 先使用本函数先取得Method,然后调用Method.invoke(Object obj, Object... args)
     */
    public static Method getAccessibleMethodByName(final Object obj, final String methodName, int argsNum) {
        // 为空不报错。直接返回 null
        if (obj == null) {
            return null;
        }
        Validate.notBlank(methodName, "methodName can't be blank");
        for (Class<?> searchType = obj.getClass(); searchType != Object.class; searchType = searchType.getSuperclass()) {
            Method[] methods = searchType.getDeclaredMethods();
            for (Method method : methods) {
                if (method.getName().equals(methodName) && method.getParameterTypes().length == argsNum) {
                    makeAccessible(method);
                    return method;
                }
            }
        }
        return null;
    }

    /**
     * 改变 private/protected 的方法为 public
     * 尽量不调用实际改动的语句
     * 避免JDK的SecurityManager抱怨
     *
     * @param method 对应的属性方法
     */
    public static void makeAccessible(Method method) {
        // 判断方法的访问修饰符
        if ((!Modifier.isPublic(method.getModifiers()) || !Modifier.isPublic(method.getDeclaringClass().getModifiers()))
                && !method.isAccessible()) {
            // 设置为公有 - 强制访问
            method.setAccessible(true);
        }
    }

    /**
     * 改变private/protected的成员变量为public
     * 尽量不调用实际改动的语句
     * 避免JDK的SecurityManager抱怨
     *
     * @param field 属性对象
     */
    public static void makeAccessible(Field field) {
        // 判断属性的访问修饰符
        if ((!Modifier.isPublic(field.getModifiers()) || !Modifier.isPublic(field.getDeclaringClass().getModifiers())
                || Modifier.isFinal(field.getModifiers())) && !field.isAccessible()) {
            // 设置为公有 - 强制访问
            field.setAccessible(true);
        }
    }

    /**
     * 通过反射, 获得 Class定义中声明的泛型参数的类型, 注意泛型必须定义在父类处
     * 如无法找到, 返回 Object.class.
     *
     * @param clazz 对象类对象
     * @return 返回 Class 定义中声明的父类的泛型参数的类型
     */
    @SuppressWarnings("unchecked")
    public static <T> Class<T> getClassGenricType(final Class clazz) {
        return getClassGenricType(clazz, 0);
    }

    /**
     * 通过反射, 获得 Class 定义中声明的父类的泛型参数的类型.
     * 如无法找到, 返回 Object.class.
     *
     * @param clazz 对象类对象
     * @param index 指定索引位置
     * @return 返回 Class 定义中声明的父类的泛型参数的类型
     */
    public static Class getClassGenricType(final Class clazz, final int index) {
        // 获取 Class 定义中声明的父类的泛型参数的类型
        Type genType = clazz.getGenericSuperclass();

        // 判断是否为泛型参数的类型
        if (!(genType instanceof ParameterizedType)) {
            logger.info("{} 的父类没有对应的泛型参数类型！", clazz.getSimpleName());
            return Object.class;
        }

        // 获取泛型参数数据集
        Type[] params = ((ParameterizedType) genType).getActualTypeArguments();

        // 判断指定索引是否在泛型参数数据集范围内
        if (index >= params.length || index < 0) {
            logger.info("指定索引 {} 不在泛型参数数据集范围内!", index);
            return Object.class;
        }

        // 泛型参数类不是 Class 类及其子类
        if (!(params[index] instanceof Class)) {
            logger.info("{} 不是 Class 类及其子类！", clazz.getSimpleName());
            return Object.class;
        }

        return (Class) params[index];
    }

    /**
     * 获取用户对应的实体对象类
     *
     * @param instance 实体对象
     * @return 返回对应的实体对象类
     */
    public static Class<?> getUserClass(Object instance) {
        // 对象判空
        if (instance == null) {
            logger.info("实体对象不能为空！");
            throw new RuntimeException("实体对象不能为空！");
        }
        // 获取对象的实体对象类
        Class clazz = instance.getClass();
        // 对象名称是否包含指定字符串 - 是否继承
        if (clazz.getName().contains(CGLIB_CLASS_SEPARATOR)) {
            // 获取对应的父类
            Class<?> superClass = clazz.getSuperclass();
            if (superClass != null && !Object.class.equals(superClass)) {
                return superClass;
            }
        }
        return clazz;
    }

    /**
     * 将反射时的 checked exception转换为 unchecked exception.
     *
     * @param msg 反射的异常信息
     * @param e 异常对象
     * @return 返回对应的异常
     */
    public static RuntimeException convertReflectionExceptionToUnchecked(String msg, Exception e) {
        if (e instanceof IllegalAccessException || e instanceof IllegalArgumentException
                || e instanceof NoSuchMethodException) {
            return new IllegalArgumentException(msg, e);
        } else if (e instanceof InvocationTargetException) {
            return new RuntimeException(msg, ((InvocationTargetException) e).getTargetException());
        }
        return new RuntimeException(msg, e);
    }
}
